package com.sat.client.utils

enum class FileType {
    DIRECTORY,
    IMAGE,
    VIDEO,
    AUDIO,
    DOCUMENT,
    OTHER
}

object FileTypeUtils {

    /**
     * 根据文件名判断文件类型
     * @param fileName 文件名
     * @return 文件类型
     */
    fun getFileType(fileName: String): FileType {
        val extension = fileName.substringAfterLast('.', "").lowercase()
        return when (extension) {
            "jpg", "jpeg", "png", "gif", "bmp", "webp" -> FileType.IMAGE
            "mp4", "mkv", "flv", "avi", "mov", "wmv", "webm" -> FileType.VIDEO
            "mp3", "wav", "flac", "aac", "ogg" -> FileType.AUDIO
            "pdf", "doc", "docx", "xls", "xlsx", "ppt", "pptx", "txt" -> FileType.DOCUMENT
            else -> FileType.OTHER
        }
    }
}